import sys
import os  # 添加os模块导入

# 获取当前脚本所在目录
current_dir = os.path.dirname(os.path.abspath(__file__))
# 添加当前目录到Python路径
sys.path.append(current_dir)

# 直接导入Game类并运行游戏
try:
    # 导入必要的模块
    import pygame
    print("pygame 模块已安装")

    # 导入Game类、Level类和SCREEN_HEIGHT常量
    from mario import Game, Level, SCREEN_HEIGHT

    # 创建并运行游戏
    print("正在启动马里奥游戏...")
    game = Game()
    
    # 添加选择关卡的选项
    print("请选择游戏模式：")
    print("1. 从第一关开始")
    print("2. 直接进入第四关")
    
    # 获取用户输入
    choice = input("请输入选择 (1 或 2): ")
    
    # 根据用户选择设置初始关卡
    if choice == "2":
        game.current_level = 4
        game.level = Level(game.current_level)
        # 重置玩家位置到第4关的出生点
        game.player.rect.x = 100
        game.player.rect.y = SCREEN_HEIGHT - 100
        game.player.velocity_x = 0
        game.player.velocity_y = 0
        game.player.on_ground = True
    else:
        # 默认从第一关开始
        print("从第一关开始游戏")
    
    game.run()

except ImportError as e:
    print(f"缺少依赖项: {e}")
    print("请运行: pip install pygame")
except Exception as e:
    print(f"启动游戏时出错: {e}")

print("游戏已退出")
